import gymnasium as gym
import highway_env

from matplotlib import pyplot as plt

env = gym.make('highway-v0', render_mode='rgb_array')

env.reset()

for _ in range(300):
    action = env.unwrapped.action_type.actions_indexes['IDLE']
    obs, reward, done, truncated, info = env.step(action)
    env.render()

plt.imshow(env.render())
plt.show()